import os
import sys
import time

script_dir_path = os.path.dirname(__file__)
project_dir_path = os.path.join(script_dir_path, "../..")
sys.path.append(project_dir_path)

from scripts.datasets.delay_repeat.dataset import make_delay_repeat_dataset_for_training

batch_samples_n = 2
vocab_tokens_n = 8
random_seed = 42

dataset = make_delay_repeat_dataset_for_training(
    batch_samples_n=batch_samples_n,
    vocab_tokens_n=vocab_tokens_n,
    random_seed=random_seed)

for x,  y, t in dataset:
    print(f"{x=}, {y=}, {t=}".replace("\n", "").replace(" ", ""))
    time.sleep(1)
